const config = require('../../../config');

// Test if potential opening or closing delimieter
// Assumes that there is a "$" at state.src[pos]
function isValidDelim(state, pos) {
    const prevChar = pos > 0 ? state.src.charCodeAt(pos - 1) : -1;
    const nextChar = pos + 1 <= state.posMax ? state.src.charCodeAt(pos + 1) : -1;

    let can_open = true, can_close = true;

    // 不能正常关闭：下个字符是数字、中英文左括号，或前字符是中英文左括号  (prevChar === 0x20/* " " */ || prevChar === 0x09/* \t */
    if ((nextChar >= 0x30/* "0" */ && nextChar <= 0x39/* "9" */) ||
            nextChar === 0x28 || nextChar === 0xFF08 ||
            prevChar === 0x28 || prevChar === 0xFF08
        ) {
        can_close = false;
    }

    // 不能正常开启：下个字符是中英文右括号或句号   nextChar === 0x20/* " " */ || nextChar === 0x09/* \t */
    if (nextChar === 0x29 || nextChar === 0xFF09 ||
        nextChar === 0x2E || nextChar === 0x3002) {
        can_open = false;
    }

    return {
        can_open: can_open,
        can_close: can_close
    };
}

function math_inline(state, silent) {
    var start, match, token, res, pos, esc_count;

    if (state.src[state.pos] !== "$") { return false; }

    res = isValidDelim(state, state.pos);
    if (!res.can_open) {
        if (!silent) { state.pending += "$"; }
        state.pos += 1;
        return true;
    }

    // First check for and bypass all properly escaped delimieters
    // This loop will assume that the first leading backtick can not
    // be the first character in state.src, which is known since
    // we have found an opening delimieter already.
    start = state.pos + 1;
    match = start;
    while ( (match = state.src.indexOf("$", match)) !== -1) {
        // Found potential $, look for escapes, pos will point to
        // first non escape when complete
        pos = match - 1;
        while (state.src[pos] === "\\") { pos -= 1; }

        // Even number of escapes, potential closing delimiter found
        if ( ((match - pos) % 2) == 1 ) { break; }
        match += 1;
    }

    // No closing delimter found.  Consume $ and continue.
    if (match === -1) {
        if (!silent) { state.pending += "$"; }
        state.pos = start;
        return true;
    }

    // Check if we have empty content, ie: $$.  Do not parse.
    if (match - start === 0) {
        if (!silent) { state.pending += "$$"; }
        state.pos = start + 1;
        return true;
    }

    // Check for valid closing delimiter
    res = isValidDelim(state, match);
    if (!res.can_close) {
        if (!silent) { state.pending += "$"; }
        state.pos = start;
        return true;
    }

    if (!silent) {
        token         = state.push('math_inline', 'math', 0);
        token.markup  = "$";
        token.content = state.src.slice(start, match);
    }

    state.pos = match + 1;
    return true;
}

function math_block(state, startLine, endLine, silent) {
    const startPos = state.bMarks[startLine] + state.tShift[startLine];
    const lineText = state.src.slice(startPos, state.eMarks[startLine]);

    // 识别两种块级公式起始标记：$$ 或 \[
    const isDollarBlock = lineText.startsWith('$$');
    const isBracketBlock = lineText.trim().startsWith('\\[');
    if (!isDollarBlock && !isBracketBlock) return false;

    // 确定结束标记
    const endMarkerRegex = isDollarBlock ? /^\s*\$\$\s*$/ : /^\s*\\]\s*$/;

    // 查找结束行
    let endMarkerLine = -1;
    for (let line = startLine + 1; line < endLine; line++) {
        const lineStart = state.bMarks[line] + state.tShift[line];
        const lineContent = state.src.slice(lineStart, state.eMarks[line]);
        if (endMarkerRegex.test(lineContent)) {
            endMarkerLine = line;
            break;
        }
    }

    if (endMarkerLine === -1) return false;

    // 提取公式内容（排除起始行和结束行）
    const content = state.getLines(startLine + 1, endMarkerLine - 1, 0, true)
        .replace(/\\\$/g, '$') // 处理转义符
        .replace(/\\\\/g, '\\');

    // 创建 Token
    const token = state.push('math_block', 'math', 0);
    token.content = content.trim();
    token.map = [startLine, endMarkerLine];
    token.markup = isDollarBlock ? '$$' : '\\\\[';
    token.block = true;

    state.line = endMarkerLine + 1;
    return true;
}
// function math_block(state, startLine, endLine, silent) {
//     // 检测起始行是否以 $$ 开头
//     const startPos = state.bMarks[startLine] + state.tShift[startLine];
//     const endPos = state.eMarks[startLine];
//     if (state.src.slice(startPos, startPos + 2) !== '$$') return false;

//     let nextLine = startLine;
//     let endMarkerLine = -1;
//     let endMarkerPos = -1;

//     // 从下一行开始寻找结束标记 $$
//     for (nextLine = startLine + 1; nextLine < endLine; nextLine++) {
//         const lineStart = state.bMarks[nextLine] + state.tShift[nextLine];
//         const lineEnd = state.eMarks[nextLine];
//         const lineContent = state.src.slice(lineStart, lineEnd);

//         // 严格匹配行首的 $$（允许缩进）
//         if (lineContent.trim() === '$$') {
//             endMarkerLine = nextLine;
//             endMarkerPos = lineStart + lineContent.indexOf('$$') + 2;
//             break;
//         }
//     }

//     // 未找到闭合标记
//     if (endMarkerLine === -1) return false;

//     if (silent) return true;

//     // 提取公式内容（排除起始行和结束行的 $$）
//     const contentStart = startPos + 2;
//     const contentEnd = state.bMarks[endMarkerLine] + state.tShift[endMarkerLine];
//     const content = state.src.slice(contentStart, contentEnd).trim();

//     // 创建公式块 Token
//     const token = state.push('math_block', 'math', 0);
//     token.content = content;
//     token.map = [startLine, endMarkerLine];
//     token.markup = '$$';
//     token.block = true;

//     // 关键！更新解析器到结束行的下一行
//     state.line = endMarkerLine + 1;
//     return true;
// }

module.exports = md => {
    var inlineRenderer = function(tokens, idx){
      return `<latex value="${encodeURIComponent(tokens[idx].content).replace(/'/g,'%27')}" type="line"></latex>`;
    };

    var blockRenderer = function(tokens, idx){
      return `<latex value="${encodeURIComponent(tokens[idx].content).replace(/'/g, '%27')}" type="block"></latex>`;
    };

    md.inline.ruler.after('escape', 'math_inline', math_inline);
    md.block.ruler.after('blockquote', 'math_block', math_block, {
        alt: [ 'paragraph', 'reference', 'blockquote', 'list' ]
    });
    md.renderer.rules.math_inline = inlineRenderer;
    md.renderer.rules.math_block = blockRenderer;
};